'''
@Project ：python-study
@File    ：8.9排序与筛选.py
@IDE     ：PyCharm
@Author  ：SUNLIN
@Date    ：2025/3/11 17:17:17
'''
import pandas as pd
import numpy as np

# 首先，获取数据，清洗后的数据
pd.set_option('display.max_rows', None)
pd.set_option('display.max_columns', None)
pd.set_option('max_colwidth', 50)

# 读取数据
data = pd.read_excel("../doc/datasource/C8-8.5-movie-data-clean.xlsx")
# 对列标签进行排序，单列/多列，正序/逆序
print(data.sort_values('title', ascending=True))
print(data.sort_values('title', ascending=False))
# 筛选： 行（索引、值、行数）、列、行列

print("筛选第0行到第3行的数据:\n", data.iloc[0: 3])  # 筛选第0行到第3行的数据
print("筛选评分等于5.5的数据:\n", data.loc[data['average'] == 5.5])  # 筛选评分等于5.5的数据

print(data.loc[data['average'] == 5.5,['genre', 'title', 'release_date']])
